package com.leetcode.no118;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Arrays;

public class Solution {
    private List<List<Integer>> res;

    public List<List<Integer>> generate(int numRows) {
        res = new ArrayList<>();
        if (numRows == 0)
            return res;

        for (int i = 1; i <= numRows; i++) {
            res.add(generateN(i));
        }
        return res;
    }

    private List<Integer> generateN(int n) {
        List<Integer> resN = new LinkedList<>();
        if (n == 1) {
            resN.add(1);
            return resN;
        }
        if (n == 2) {
            resN.add(1);
            resN.add(1);
            return resN;
        }
        if (n == 3) {
            resN.add(1);
            resN.add(2);
            resN.add(1);
            return resN;
        }

        List<Integer> tmp;

        // 返回第n-1行的数据
        tmp = res.get(n - 2);

        LinkedList<Integer>tmp01 = new LinkedList<Integer>();
        LinkedList<Integer>tmp02 = new LinkedList<Integer>();

        tmp01.add(0);
        for (int i = 0; i < n; i++) {
            int num = tmp.get(i);
            tmp01.add(num);
            tmp02.add(num);
        }
        tmp02.add(0);


//        for (int i = 0; i < n+1; i++) {
//            resN.add(tmp01.get(i) + tmp02.get(i));
//        }

        while(!tmp01.isEmpty() && !tmp02.isEmpty()){
            resN.add(tmp01.pollFirst() + tmp02.pollFirst());
        }

        return resN;
    }
}


class Solution02 {
    public List<List<Integer>> generate(int numRows) {
        List<List<Integer>> list = new ArrayList<>();
        if (numRows == 0){
            return list;
        }
        Integer[] arr = {1};
        list.add(Arrays.asList(arr));
        int[] rootarr = {1};
        for (int i = 1; i < numRows; i++) {
            rootarr = arrcount(rootarr,i);
            Integer[] mid = new Integer[rootarr.length];
            for (int j = 0; j < rootarr.length; j++) {
                mid[j] = rootarr[j];
            }
            list.add(Arrays.asList(mid));
        }
        return list;
    }

    public int[] arrcount(int[] arr,int n){
        int[] arr1 = new int[n + 1];
        int[] arr2 = new int[n + 1];
        for (int i = 0; i < arr.length; i++) {
            arr1[i] = arr[i];
            arr2[i+1] = arr[i];
        }
        int[] arr3 = new int[n+1];
        for (int i = 0; i < arr1.length; i++) {
            arr3[i] = arr1[i] + arr2[i];
        }
        return arr3;
    }
}